Skip to content

feat: add tfds dataset format and bridge v2 dataset#400

Open
kwangneuraco wants to merge 1 commit intodevelopfrom
feat/add_tfds_bridge_dataset
Open

feat: add tfds dataset format and bridge v2 dataset#400
kwangneuraco wants to merge 1 commit intodevelopfrom
feat/add_tfds_bridge_dataset

Conversation

@kwangneuraco
Copy link
Contributor

  • Add tfds support, which is a alive dataformat and slightly better than rlds (stopped maintaining)
  • add bridge v2 dataset import config
  • add tests relevant and lerobot test

@github-actions
Copy link
Contributor

github-actions bot commented Feb 13, 2026

PR source branch is valid

  • Source: feat/add_tfds_bridge_dataset
  • Target: develop

@kwangneuraco kwangneuraco added the version:minor non-breaking feature updates, new functions or endpoints label Feb 13, 2026
@github-actions
Copy link
Contributor

Consider updating changelogs/pending-changelog.md with a summary of this change for the release notes. This is optional and non-blocking.

@kwangneuraco kwangneuraco force-pushed the feat/add_tfds_bridge_dataset branch from 90100d4 to 9270d2a Compare February 13, 2026 12:15
@sdas-neuraco
Copy link
Contributor

Split into two PRs with Lerobot tests in a separate one. We should back test the existing Datasets to check nothing is broken with RLDS. comments included from @ypang-neuraco

@kwangneuraco kwangneuraco force-pushed the feat/add_tfds_bridge_dataset branch from 9270d2a to 22bbf62 Compare February 13, 2026 13:37
Copy link
Contributor

@ypang-neuraco ypang-neuraco left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please check if you can merge the _record_step of TFDS and RLDS. It seems to be the same to me.

@kwangneuraco kwangneuraco force-pushed the feat/add_tfds_bridge_dataset branch 3 times, most recently from 41b0031 to db4978c Compare February 18, 2026 16:33
Copy link
Contributor

@ypang-neuraco ypang-neuraco left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM \w comments. Please remove the old code for RLDS importer if it is not used anymore.

Comment on lines +430 to +435
elif (
self.allow_mapping_name_fallback
and isinstance(source, dict)
and item.name in source
):
source_data = source[item.name]
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please remove the fallback on item name, users should fix the source name if it is incorrect.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

version:minor non-breaking feature updates, new functions or endpoints

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants

Comments